﻿// Gateway.Api 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
//
// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
//
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！

using Aurora.AI.Abstractions;
using Aurora.AI.Abstractions.Images;
using Aurora.AI.Abstractions.Images.Dtos;
using Aurora.AI.Abstractions.ObjectModels.RequestModels;
using Aurora.AI.Abstractions.ObjectModels.ResponseModels.ImageResponseModel;
using Aurora.AI.MetaGLM.Models.RequestModels;
using System.Text;

namespace Aurora.AI.MetaGLM.Images;
public class MetaGLMImageService(Modules.Images images) : IAuroraAIImageService
{
    /// <summary>
    /// 创建文生图
    /// </summary>
    /// <param name="imageCreate"></param>
    /// <param name="options"></param>
    /// <param name="cancellationToken"></param>
    /// <returns></returns>
    public Task<ImageCreateResponse> CreateImage(AuroraAIImageCreateRequest imageCreate, AuroraAIPlatformOptions options = null, CancellationToken cancellationToken = default)
    {
        StringBuilder sb = new StringBuilder();
        ImageRequestBase request = new ImageRequestBase();
        request.SetModel(imageCreate.Model);
        request.SetPrompt(imageCreate.Prompt);
        var response = images.Generation(request, options.ApiKey);
        ImageCreateResponse result = new ImageCreateResponse();

        // 证明返回正常
        if (response?.error?.Count > 0)
        {
            response.error.TryGetValue("code", out var code);
            response.error.TryGetValue("message", out var message);
            result.Error = new Abstractions.Dtos.AuroraAIError();
            result.Error.Code = code;
            result.Error.Messages = new List<string> { message };
            return Task.FromResult(result);
        }
        result.Results = new List<ImageCreateResponse.ImageDataResult>();
        foreach (var item in response.data)
        {
            result.Results.Add(new ImageCreateResponse.ImageDataResult { Url = item.url });
        }
        return Task.FromResult(result);
    }

    public Task<ImageCreateResponse> CreateImageEdit(AuroraAIImageEditCreateRequest imageEditCreateRequest, AuroraAIPlatformOptions options = null, CancellationToken cancellationToken = default)
    {
        throw new NotImplementedException();
    }

    public Task<ImageCreateResponse> CreateImageVariation(AuroraAIImageVariationCreateRequest imageEditCreateRequest, AuroraAIPlatformOptions options = null, CancellationToken cancellationToken = default)
    {
        throw new NotImplementedException();
    }
}
